from fastapi import APIRouter, HTTPException, status
from sqlalchemy import select

from tracecat.auth.dependencies import WorkspaceUserRole
from tracecat.db.dependencies import AsyncDBSession
from tracecat.db.models import Schedule
from tracecat.exceptions import TracecatNotFoundError, TracecatServiceError
from tracecat.identifiers.workflow import OptionalAnyWorkflowIDQuery, WorkflowUUID
from tracecat.logger import logger
from tracecat.workflow.management.management import WorkflowsManagementService
from tracecat.workflow.schedules.dependencies import AnyScheduleIDPath
from tracecat.workflow.schedules.schemas import (
    ScheduleCreate,
    ScheduleRead,
    ScheduleSearch,
    ScheduleUpdate,
)
from tracecat.workflow.schedules.service import WorkflowSchedulesService

router = APIRouter(prefix="/schedules", tags=["schedules"])


@router.get("", response_model=list[ScheduleRead])
async def list_schedules(
    role: WorkspaceUserRole,
    session: AsyncDBSession,
    workflow_id: OptionalAnyWorkflowIDQuery,
) -> list[ScheduleRead]:
    service = WorkflowSchedulesService(session, role=role)
    schedules = await service.list_schedules(workflow_id=workflow_id)
    return ScheduleRead.list_adapter().validate_python(schedules)


@router.post("", response_model=ScheduleRead)
async def create_schedule(
    role: WorkspaceUserRole,
    session: AsyncDBSession,
    params: ScheduleCreate,
) -> ScheduleRead:
    """Create a schedule for a workflow."""
    service = WorkflowSchedulesService(session, role=role)
    workflow_svc = WorkflowsManagementService(session, role=role)
    workflow = await workflow_svc.get_workflow(WorkflowUUID.new(params.workflow_id))
    if not workflow:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Workflow not found. Please check the workflow ID and try again.k",
        )
    if not workflow.version:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Workflow must be saved before creating a schedule.",
        )
    try:
        schedule = await service.create_schedule(params)
        return ScheduleRead.model_validate(schedule)
    except TracecatServiceError as e:
        logger.error("Error creating schedule", error=e)
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Error creating schedule: {e}. Please try again or contact support.",
        ) from e


@router.get("/{schedule_id}", response_model=ScheduleRead)
async def get_schedule(
    role: WorkspaceUserRole,
    session: AsyncDBSession,
    schedule_id: AnyScheduleIDPath,
) -> ScheduleRead:
    """Get a schedule from a workflow."""
    service = WorkflowSchedulesService(session, role=role)
    try:
        schedule = await service.get_schedule(schedule_id)
        return ScheduleRead.model_validate(schedule)
    except TracecatNotFoundError as e:
        logger.error("Error getting schedule", error=e)
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=f"Schedule {schedule_id} not found. Please check whether this schedule exists and try again.",
        ) from e


@router.post("/{schedule_id}", response_model=ScheduleRead)
async def update_schedule(
    role: WorkspaceUserRole,
    session: AsyncDBSession,
    schedule_id: AnyScheduleIDPath,
    params: ScheduleUpdate,
) -> ScheduleRead:
    """Update a schedule from a workflow. You cannot update the Workflow Definition, but you can update other fields."""
    service = WorkflowSchedulesService(session, role=role)
    try:
        schedule = await service.update_schedule(schedule_id, params)
        return ScheduleRead.model_validate(schedule)
    except TracecatNotFoundError as e:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=f"Schedule {schedule_id} not found. Please check whether this schedule exists and try again.",
        ) from e
    except TracecatServiceError as e:
        logger.error("Error updating schedule", error=e)
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Failed to update schedule: {e}. Please try again or contact support.",
        ) from e


@router.delete("/{schedule_id}", status_code=status.HTTP_204_NO_CONTENT)
async def delete_schedule(
    role: WorkspaceUserRole, session: AsyncDBSession, schedule_id: AnyScheduleIDPath
) -> None:
    """Delete a schedule from a workflow."""
    service = WorkflowSchedulesService(session, role=role)
    try:
        await service.delete_schedule(schedule_id)
    except TracecatServiceError as e:
        logger.error("Error deleting schedule", error=e)
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Failed to delete schedule: {e}. Please try again or contact support.",
        ) from e


@router.get("/search", response_model=list[ScheduleRead])
async def search_schedules(
    role: WorkspaceUserRole, session: AsyncDBSession, params: ScheduleSearch
) -> list[ScheduleRead]:
    """**[WORK IN PROGRESS]** Search for schedules."""
    statement = select(Schedule).where(Schedule.workspace_id == role.workspace_id)
    results = await session.execute(statement)
    schedules = results.scalars().all()
    return ScheduleRead.list_adapter().validate_python(schedules)
